from flask import g

"""
自定义一个装饰器，验证某些请求是否已经登陆过了，如果已经登陆录了继续访问。
本质上是一个登陆的拦截器
"""


def login_required(func):
    def wrapper(*args, **kwargs):
        if g.user_id is not None:  # 已经登陆过了
            return func(*args, **kwargs)
        else:
            return {'massage': 'Invalid Token.'}, 400

    return wrapper
